#pragma once
#include "Comm.hpp"
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <iostream>
#include <string>
#include <cstring>
#include <cstdio>
#include <unistd.h>
const int gdefaultid = -1;
const int gsize = 4096;
const std::string pathname = ".";
const int projid = 0x66;
const int gmode = 0666;
#define CREATER "creater"
#define USER "user"

class Shm
{
private:
    // 我们的创建共享内存和获取共享内存是只有一字之差的
    // IPC_CREAT | IPC_EXCL | gmode
    // 获取是不需要后面两个的

    // 创建的一定要是一个全新的共享内存
    void CreatHelper(int flg) // 创建共享内存
    {

        printf("key:0x%x\n", _key);
        // 共享内促的生命周期，随内核，我们必须使用手动或者代码进行删除操作
        //_shmid = shmget(k, _size, IPC_CREAT | IPC_EXCL | gmode); // 利用第三个参数保证是全新的共享内存
        _shmid = shmget(_key, _size, flg); // 利用第三个参数保证是全新的共享内存

        // Linux下一切皆文件，我们第三个参数后面加上了Mode，大小是0666，就是权限
        if (_shmid < 0) // 创建失败了
        {
            ERR_EXIT("shmget");
        }
        printf("shmid:%d", _shmid);
    }
    void Creat() // 创建共享内存
    {
        CreatHelper(IPC_CREAT | IPC_EXCL | gmode); // 创建共享内存
    }
    void Attach() // 让进程和共享内存进行关联操作
    {
        _start_mem = shmat(_shmid, nullptr, 0); // 将共享内存与当前进程进行关联，第三个参数为0表示让系统自动分配内存
        if ((long long)_start_mem < 0)          // 关联失败了
        {
            ERR_EXIT("shmat\n");
        }
        printf("attach success\n");
        // 关联成功了
    }

    void Detach() // 让进程和共享内存进行解除关联操作
    {
        int n = shmdt(_start_mem); // 解除关联操作
        if (n == 0)                // 如果成功的话返回值是0
        {
            printf("detach success\n");
        }
    }

    void Get() // 获取共享内存
    {
        CreatHelper(IPC_CREAT); // 创建共享内存
    }
    void Destroy()
    {
        // if (_shmid == gdefaultid)
        //     return; // 如果这个id==-1的话，说明我们的共享内存是没有被建立的，那么我们直接返回就行了
        // 不管是那个进程，都要在析构的时候调用detch去关联
        Detach(); // 让进程和共享内存进行解除关联操作
        // 到这里的话就说明我们是建立了共享内存的
        if (_usertype == CREATER)
        {
            int n = shmctl(_shmid, IPC_RMID, nullptr); // 删除共享内存,我们是不关心属性的，所以我们这里第三个参数设置为空
            // IPC_RMID 表明要删除指定的共享内存段。
            if (n > 0) // 删除成功了
            {
                printf("shmctl delete shm :%d success\n", _shmid);
            }
            else // 删除失败了
            {
                ERR_EXIT("shmctl\n");
            }
        }
    }

public:
    Shm(const std::string &pathname, int projid, const std::string &usertype)
        : _shmid(gdefaultid), _size(gsize), _start_mem(nullptr), _usertype(usertype)
    {
        _key = ftok(pathname.c_str(), projid); // 创建key值
        if (_key < 0)                          // 创建失败了
        {
            ERR_EXIT("ftok\n");
        }
        // 根据用户的类型我们进行共享内存的操作
        // 如果是创建者的话就进行共享内存的创建操作，如果是使用这的话那么就进行获取共享内存的操作就行了
        if (usertype == CREATER)
            Creat(); // 创建共享内存
        else if (usertype == USER)
            Get(); // 获取共享内存
        else
        {
        }

        Attach(); // 让进程和共享内存进行关联操作
    }
    void *VirtualAddr()
    {
        printf("VirtualAddr:%p\n", _start_mem);
        return _start_mem;
    }
    int size()
    {
        return _size;
    }

    //或者指定共享内存的属性
    void Attr()
    {
        struct shmid_ds ds;
        //IPC_STAT是状态
        //ds是输出型参数，用来存放共享内存的属性
        int n=shmctl(_shmid, IPC_STAT, &ds); // 获取共享内存的属性
        printf("shm_segsz:%ld\n",ds.shm_segsz);
        printf("key:0x%x\n",ds.shm_perm.__key);
    }
    ~Shm()
    {
        // 只有创建者才需要进行析构操作
        if (_usertype == CREATER)
            Destroy();
    }

private:
    int _shmid;
    key_t _key;
    int _size;
    void *_start_mem; // 共享内存的起始地址
    std::string _usertype;
};